package com.sky.handler;

import com.sky.constant.MessageConstant;
import com.sky.exception.BaseException;
import com.sky.result.Result;
import lombok.extern.slf4j.Slf4j;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

/**
 * 全局异常处理器，处理项目中抛出的业务异常和数据库异常
 */
@RestControllerAdvice
@Slf4j
public class GlobalExceptionHandler {

    /**
     * 捕获业务异常
     * @param ex 业务异常对象
     * @return 处理结果对象
     */
    @ExceptionHandler
    public Result handleBusinessException(BaseException ex){
        log.error("业务异常信息：{}", ex.getMessage());
        return Result.error(ex.getMessage());
    }

    /**
     * 捕获数据库唯一键约束异常
     * @param ex 唯一键约束异常对象
     * @return 处理结果对象
     */
    @ExceptionHandler(DuplicateKeyException.class)
    public Result handleDuplicateKeyException(DuplicateKeyException ex){
        log.error("数据库唯一键约束异常");
        // 从异常消息中提取重复键值
        String duplicateKey = extractDuplicateKey(ex.getMessage());
        String errorMessage = duplicateKey + MessageConstant.ALREADY_EXISTS;
        return Result.error(errorMessage);
    }

    /**
     * 从异常消息中提取重复键值
     * @param message 异常消息
     * @return 重复键值
     */
    private String extractDuplicateKey(String message){
        // 从异常消息中提取重复键值的逻辑，请根据实际情况修改
        // 示例中简单地使用字符串分割获取重复键值
        String[] parts = message.split("重复键值为 \\(");
        if(parts.length > 1) {
            String keyPart = parts[1];
            return keyPart.substring(0, keyPart.indexOf(")"));
        }
        return "Unknown";
    }

    /**
     * 捕获未知异常
     * @param ex 未知异常对象
     * @return 处理结果对象
     */
    @ExceptionHandler(Exception.class)
    public Result handleUnknownException(Exception ex){
        log.error("未知异常：{}", ex.getMessage(), ex);
        return Result.error(MessageConstant.UNKNOWN_ERROR);
    }
}
